;;; guile-hkdf --- HKDF: HMAC based Extract-and-Expand Key Derivation Function
;;; Copyright © 2020 Brendan Tildesley <mail@brendan.scot>
;;;
;;; This file is part of guile-hkdf.
;;;
;;; guile-hkdf is free software; you can redistribute it and/or modify it
;;; under the terms of the GNU General Public License as published by
;;; the Free Software Foundation; either version 3 of the License, or
;;; (at your option) any later version.
;;;
;;; guile-hkdf is distributed in the hope that it will be useful, but
;;; WITHOUT ANY WARRANTY; without even the implied warranty of
;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
;;; General Public License for more details.
;;;
;;; You should have received a copy of the GNU General Public License
;;; along with guile-gcrypt.  If not, see <http://www.gnu.org/licenses/>.

(define-module (test-hkdf)
  #:use-module (srfi srfi-64)
  #:use-module (rnrs bytevectors)
  #:use-module (guix base16)
  #:use-module (hkdf))

(test-begin "HKDF")

(test-equal "A.1. Test Case 1. hkdf-extract with SHA-256"
  (base16-string->bytevector
   "077709362c2e32df0ddc3f0dc47bba6390b6c73bb50f9c3122ec844ad7c2b3e5")
  (hkdf-extract
   (base16-string->bytevector
    "0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b")
   #:salt (base16-string->bytevector "000102030405060708090a0b0c")
   #:algorithm (lookup-mac-algorithm 'hmac-sha256)))

(test-equal "A.1. Test Case 1. hkdf-expand with SHA-256 (completing full HKDF)"
  (base16-string->bytevector
   "3cb25f25faacd57a90434f64d0362f2a2d2d0a90cf1a5a4c5db02d56ecc4c5bf34007208d5b887185865")
  (hkdf-expand
   (base16-string->bytevector
    "077709362c2e32df0ddc3f0dc47bba6390b6c73bb50f9c3122ec844ad7c2b3e5")

   #:info (base16-string->bytevector "f0f1f2f3f4f5f6f7f8f9")
   #:length  42
   #:algorithm (lookup-mac-algorithm 'hmac-sha256)))

(test-equal "A.2. Test Case 2. hkdf-extract longer inputs/outputs with SHA256"
  (base16-string->bytevector
   "06a6b88c5853361a06104c9ceb35b45cef760014904671014a193f40c15fc244")
  (hkdf-extract
   (base16-string->bytevector
    (string-append
     "000102030405060708090a0b0c0d0e0f"
     "101112131415161718191a1b1c1d1e1f"
     "202122232425262728292a2b2c2d2e2f"
     "303132333435363738393a3b3c3d3e3f"
     "404142434445464748494a4b4c4d4e4f"))
   #:salt (base16-string->bytevector
           (string-append
            "606162636465666768696a6b6c6d6e6f7071727374757"
            "67778797a7b7c7d7e7f808182838485868788898a8b8c"
            "8d8e8f909192939495969798999a9b9c9d9e9fa0a1a2a"
            "3a4a5a6a7a8a9aaabacadaeaf"))
   #:algorithm (lookup-mac-algorithm 'hmac-sha256)))


(test-equal "A.2. Test Case 2. hkdf-expand longer inputs/outputs with SHA256"
  (base16-string->bytevector
   (string-append
    "b11e398dc80327a1c8e7f78c596a4934"
    "4f012eda2d4efad8a050cc4c19afa97c"
    "59045a99cac7827271cb41c65e590e09"
    "da3275600c2f09b8367793a9aca3db71"
    "cc30c58179ec3e87c14c01d5c1f3434f1d87"))

  (hkdf-expand
   (base16-string->bytevector
    "06a6b88c5853361a06104c9ceb35b45cef760014904671014a193f40c15fc244")
   #:info (base16-string->bytevector
           (string-append
            "b0b1b2b3b4b5b6b7b8b9babbbcbdbebf"
            "c0c1c2c3c4c5c6c7c8c9cacbcccdcecf"
            "d0d1d2d3d4d5d6d7d8d9dadbdcdddedf"
            "e0e1e2e3e4e5e6e7e8e9eaebecedeeef"
            "f0f1f2f3f4f5f6f7f8f9fafbfcfdfeff"))
   #:length 82
   #:algorithm (lookup-mac-algorithm 'hmac-sha256)))




(test-equal "A.2. Test Case 2. hkdf-extract longer inputs/outputs with SHA256 (hkdf function)"
  (base16-string->bytevector
   (string-append
    "b11e398dc80327a1c8e7f78c596a4934"
    "4f012eda2d4efad8a050cc4c19afa97c"
    "59045a99cac7827271cb41c65e590e09"
    "da3275600c2f09b8367793a9aca3db71"
    "cc30c58179ec3e87c14c01d5c1f3434f1d87"))

  (hkdf
   (base16-string->bytevector
    (string-append
     "000102030405060708090a0b0c0d0e0f"
     "101112131415161718191a1b1c1d1e1f"
     "202122232425262728292a2b2c2d2e2f"
     "303132333435363738393a3b3c3d3e3f"
     "404142434445464748494a4b4c4d4e4f"))
   #:salt (base16-string->bytevector
           (string-append
            "606162636465666768696a6b6c6d6e6f7071727374757"
            "67778797a7b7c7d7e7f808182838485868788898a8b8c"
            "8d8e8f909192939495969798999a9b9c9d9e9fa0a1a2a"
            "3a4a5a6a7a8a9aaabacadaeaf"))
   #:info (base16-string->bytevector
           (string-append
            "b0b1b2b3b4b5b6b7b8b9babbbcbdbebf"
            "c0c1c2c3c4c5c6c7c8c9cacbcccdcecf"
            "d0d1d2d3d4d5d6d7d8d9dadbdcdddedf"
            "e0e1e2e3e4e5e6e7e8e9eaebecedeeef"
            "f0f1f2f3f4f5f6f7f8f9fafbfcfdfeff"))
   #:length 82
   #:algorithm (lookup-mac-algorithm 'hmac-sha256)))

(test-equal "A.3. Hkdf-Extract with zero-length salt/info with SHA256"
  (base16-string->bytevector
   "19ef24a32c717b167f33a91d6f648bdf96596776afdb6377ac434c1c293ccb04")

  (hkdf-extract
   (base16-string->bytevector
    "0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b")
   #:salt #vu8()
   #:algorithm (lookup-mac-algorithm 'hmac-sha256)))

(test-equal "A.3. Hkdf-Extract with zero-length salt/info with SHA256 -- Unspecified
salt defaults to zero-length"
  (base16-string->bytevector
   "19ef24a32c717b167f33a91d6f648bdf96596776afdb6377ac434c1c293ccb04")

  (hkdf-extract
   (base16-string->bytevector
    "0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b")
   #:algorithm (lookup-mac-algorithm 'hmac-sha256)))

(test-equal "A.3. Hkdf-Expand zero-length salt/info with SHA256"
  (base16-string->bytevector
   "8da4e775a563c18f715f802a063c5a31b8a11f5c5ee1879ec3454e5f3c738d2d9d201395faa4b61a96c8")

  (hkdf-expand
   (base16-string->bytevector
    "19ef24a32c717b167f33a91d6f648bdf96596776afdb6377ac434c1c293ccb04")
   #:info #vu8()
   #:length 42
   #:algorithm (lookup-mac-algorithm 'hmac-sha256)))

(test-equal "A.4. Basic Hkdf-Extract test case with SHA-1"
  (base16-string->bytevector "9b6c18c432a7bf8f0e71c8eb88f4b30baa2ba243")

  (hkdf-extract
   (base16-string->bytevector "0b0b0b0b0b0b0b0b0b0b0b")
   #:salt (base16-string->bytevector "000102030405060708090a0b0c")
   #:algorithm (lookup-mac-algorithm 'hmac-sha1)))

(test-equal "A.4. Basic Hkdf-Expand test case with SHA-1"
  (base16-string->bytevector
   "085a01ea1b10f36933068b56efa5ad81a4f14b822f5b091568a9cdd4f155fda2c22e422478d305f3f896")

  (hkdf-expand
   (base16-string->bytevector "9b6c18c432a7bf8f0e71c8eb88f4b30baa2ba243")
   #:info (base16-string->bytevector "f0f1f2f3f4f5f6f7f8f9")
   #:length 42
   #:algorithm (lookup-mac-algorithm 'hmac-sha1)))

(test-equal "A.5. Test Case 2. hkdf-extract longer inputs/outputs with SHA1"
  (base16-string->bytevector
   "8adae09a2a307059478d309b26c4115a224cfaf6")
  (hkdf-extract
   (base16-string->bytevector
    (string-append
     "000102030405060708090a0b0c0d0e0f"
     "101112131415161718191a1b1c1d1e1f"
     "202122232425262728292a2b2c2d2e2f"
     "303132333435363738393a3b3c3d3e3f"
     "404142434445464748494a4b4c4d4e4f"))
   #:salt (base16-string->bytevector
           (string-append
            "606162636465666768696a6b6c6d6e6f7071727374757677787"
            "97a7b7c7d7e7f808182838485868788898a8b8c8d8e8f909192"
            "939495969798999a9b9c9d9e9fa0a1a2a3a4a5a6a7a8a9aaabacadaeaf"))
   #:algorithm (lookup-mac-algorithm 'hmac-sha1)))


(test-equal "A.5. Test Case 2. hkdf-expand longer inputs/outputs with SHA1"
  (base16-string->bytevector
   (string-append
    "0bd770a74d1160f7c9f12cd5912a06eb"
    "ff6adcae899d92191fe4305673ba2ffe"
    "8fa3f1a4e5ad79f3f334b3b202b2173c"
    "486ea37ce3d397ed034c7f9dfeb15c5e"
    "927336d0441f4c4300e2cff0d0900b52"
    "d3b4"))

  (hkdf-expand
   (base16-string->bytevector
    "8adae09a2a307059478d309b26c4115a224cfaf6")
   #:info (base16-string->bytevector
           (string-append
            "b0b1b2b3b4b5b6b7b8b9babbbcbdbebf"
            "c0c1c2c3c4c5c6c7c8c9cacbcccdcecf"
            "d0d1d2d3d4d5d6d7d8d9dadbdcdddedf"
            "e0e1e2e3e4e5e6e7e8e9eaebecedeeef"
            "f0f1f2f3f4f5f6f7f8f9fafbfcfdfeff"))
   #:length 82
   #:algorithm (lookup-mac-algorithm 'hmac-sha1)))

(test-equal "A.6. Hkdf-Extract with zero-length salt/info with SHA1"
  (base16-string->bytevector "da8c8a73c7fa77288ec6f5e7c297786aa0d32d01")

  (hkdf-extract
   (base16-string->bytevector
    "0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b")
   #:salt #vu8()
   #:algorithm (lookup-mac-algorithm 'hmac-sha1)))

(test-equal "A.6. Hkdf-Extract with zero-length salt/info with SHA1 -- Unspecified
salt defaults to zero-length"
  (base16-string->bytevector "da8c8a73c7fa77288ec6f5e7c297786aa0d32d01")

  (hkdf-extract
   (base16-string->bytevector
    "0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b")
   #:algorithm (lookup-mac-algorithm 'hmac-sha1)))

(test-equal "A.6. Hkdf-Expand zero-length salt/info with SHA1"
  (base16-string->bytevector
   "0ac1af7002b3d761d1e55298da9d0506b9ae52057220a306e07b6b87e8df21d0ea00033de03984d34918")

  (hkdf-expand
   (base16-string->bytevector "da8c8a73c7fa77288ec6f5e7c297786aa0d32d01")
   #:info #vu8()
   #:length 42
   #:algorithm (lookup-mac-algorithm 'hmac-sha1)))


(test-equal "A.7 Hkdf-Extract test with SHA-1, salt not provided (defaults to HashLen
zero octets), zero-length info"
  (base16-string->bytevector
   "2adccada18779e7c2077ad2eb19d3f3e731385dd")
  (hkdf-extract
   (base16-string->bytevector
    "0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c")
   #:algorithm (lookup-mac-algorithm 'hmac-sha1)))


(test-equal "A.7 Hkdf-Expand test with SHA-1, salt not provided (defaults to HashLen
zero octets), zero-length info"
  (base16-string->bytevector
   "2c91117204d745f3500d636a62f64f0ab3bae548aa53d423b0d1f27ebba6f5e5673a081d70cce7acfc48")

  (hkdf-expand
   (base16-string->bytevector "2adccada18779e7c2077ad2eb19d3f3e731385dd")
   #:length 42
   #:info #vu8()
   #:algorithm (lookup-mac-algorithm 'hmac-sha1)))
